# 调用pytorch
import torch

# torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None)
# 示例
# 设置打印选项，precision为打印小数点后的位数，threshold为打印的最大元素个数，edgeitems为打印矩阵的边缘元素个数，linewidth为打印的行宽，profile为打印的配置文件

torch.set_printoptions(precision=4, threshold=1000, edgeitems=3, linewidth=120, profile='default')
a = torch.randn(4, 4)
print(a)

# 输出
# tensor([[ 0.0140,  0.0140,  0.0140,  0.0140],
#         [ 0.0140,  0.0140,  0.0140,  0.0140],
#         [ 0.0140,  0.0140,  0.0140,  0.0140],
#         [ 0.0140,  0.0140,  0.0140,  0.0140]]) 0.0140为示例，实际值会不同

# 设置打印选项后，打印的tensor会按照设置的参数进行打印
# 如果不设置打印选项，则默认打印所有元素
# 如果设置打印选项，则只会打印部分元素
# 可以通过torch.set_printoptions()函数来设置打印选项，包括精度、阈值、边缘元素个数、行宽和配置文件等参数。

# tensor([[-2.7702e-01,  1.2808e-03, -1.3529e+00,  9.6112e-01],
#       [-9.2037e-01,  1.0604e+00,  1.0315e+00, -7.6124e-01],
#        [ 7.0717e-01, -1.5903e+00,  2.1037e+00, -7.3477e-01],
#        [-2.6129e-01, -1.0671e+00, -2.0468e-01,  1.0095e+00]])
# 输出出现不正确情况，在一次运行时出现科学计数法，另一次运行时没有出现科学计数法，原因不明